import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))

from app import create_app, db
from app.models import User

def create_admin():
    app = create_app()
    with app.app_context():
        # 检查管理员是否已存在
        admin = User.query.filter_by(username='admin').first()
        if admin:
            print('管理员账号已存在')
            return
        
        # 创建管理员账号
        admin = User(
            username='admin',
            email='admin@example.com',
            role='manager',
            full_name='系统管理员',
            department='系统管理部',
            phone='13800000000'
        )
        admin.set_password('admin123')
        
        try:
            db.session.add(admin)
            db.session.commit()
            print('管理员账号创建成功')
            print('用户名: admin')
            print('密码: admin123')
        except Exception as e:
            db.session.rollback()
            print('创建管理员账号失败:', str(e))

if __name__ == '__main__':
    create_admin() 